import matplotlib.pyplot as plt

# 15.2.8 自动保存图表
# 将plt.show()替换为plt.savefig()
x_values = range(1, 1001)
y_values = [x**2 for x in x_values]

plt.style.use('seaborn')
fig, ax = plt.subplots()

# 更多的colormap reference示例：https://matplotlib.org/stable/gallery/color/colormap_reference.html
ax.scatter(x_values, y_values, c=y_values, cmap=plt.cm.coolwarm, s=10)

# 设置标题坐标轴
ax.set_title('平方数', fontsize=24)
ax.set_xlabel('值', fontsize=14)
ax.set_ylabel('值的平方', fontsize=14)

# 设置每个坐标轴的取值范围
# axis()的更多用法：https://blog.csdn.net/mighty13/article/details/113851709
ax.axis([0, 1100, 0, 1100000])

# 设置中文字体
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
# plt.show()

# 更多参数：https://matplotlib.org/stable/api/_as_gen/matplotlib.pyplot.savefig.html?highlight=savefig
# 中文：https://blog.csdn.net/wilbeok/article/details/111194473
# 保存文件的路径调用python时的路径相关。即调用时的路径为工作路径。
plt.savefig('./images/colormap.png', bbox_inches='tight')
